import pymysql


class DB:
    def __init__(self):
        #连接数据库
        self.coon=pymysql.connect(host='localhost',port=3306,user='root',password='root',db='workf',charset='utf8',cursorclass=pymysql.cursors.DictCursor)
        #创建游标
        self.cousor=self.coon.cursor()

    #添加
    def update(self,sql):
        #调用游标提交sql语句
        self.cousor.execute(sql)
        self.coon.commit()
    #查询单个
    def find(self,sql):
        #调用游标提交sql
        self.cousor.execute(sql)
        #查找单个
        res=self.cousor.fetchone()
        return res
    # 查询所有
    def findall(self,sql):
        #调用游标提交sql语句
        self.cousor.execute(sql)
        #查所有
        res=self.cousor.fetchall()
        #返回
        return res

    #关闭数据库
    def close(self):
        #关闭游标
        self.cousor.close()
        #关闭连接
        self.coon.close()

db1=DB()


